rainbow in python programming